rflow (version 0.0.0.9000)

TensorFlow DNN Regressor: TensorFlow DNN Regressor

Description

Deep Neural Network regressor

Usage

TensorFlowDNNRegressor(hidden_units, ...)

Arguments

hidden_units

A vector of the number of hidden units in each layer

...

Additional parameters such as keep_prob. A list can be found here: https://github.com/google/skflow/blob/master/skflow/ops/dnn_ops.py